import os
from main import fun_caller_dict
import shutil
import pandas as pd

topic_name = '物流控制塔简介'
father_dir = '生生ESG介绍'        # 选用的模板
pages = [
    '名词解释-物流控制塔',
    '名词解释-信息共享中心',
    '时间线-测试11111111',
    '三级主题排版1-测试222222',
    '要点六边形-测试33333',
    '链接-测试4',
    '截图-测试5555',
]












# -------------------------------------------------重命名
if os.path.exists(f'templates/{topic_name}') is False:
    shutil.copytree(f'templates/{father_dir}', f'templates/{topic_name}')

img_dir = f'templates/{topic_name}/imgs'
try:
    shutil.rmtree(img_dir)
    os.makedirs(img_dir)
except Exception as e:
    print(e)

# -------------------------------------------------重命名
try:
    os.remove(f'templates/{topic_name}/output.pptx')
    os.rename(f'templates/{topic_name}/PAGES.xlsx', f'templates/{topic_name}/PAGES模板参考.xlsx')
except Exception as e:
    # print(f'重命名文件时出错: {e}')
    pass


# 创建一个ExcelWriter对象，并指定文件名
with pd.ExcelWriter(f'templates/{topic_name}/PAGES删除中文后编辑.xlsx') as writer:
    current_t = list(fun_caller_dict.keys())  # 现有的可处理模板
    for page in pages:
        if page.split('-')[0] not in current_t:
            print('没找到模板------------------->', page.split('-')[0])
            break  
        if page.split('-')[0]=='名词解释':
            df = pd.DataFrame([], columns='解释,关键词'.split(','))
        if page.split('-')[0]=='时间线':
            df = pd.DataFrame([], columns='时间,主体,事件'.split(','))
        if page.split('-')[0]=='三级主题排版1':
            df = pd.DataFrame([], columns='c1,c2,c3'.split(','))
        if page.split('-')[0]=='要点':
            df = pd.DataFrame([], columns='要点'.split(','))
        if page.split('-')[0]=='要点六边形':
            df = pd.DataFrame([], columns='要点'.split(','))
        if page.split('-')[0]=='链接':
            df = pd.DataFrame([], columns='名称,链接'.split(','))
        if page.split('-')[0]=='截图':
            df = pd.DataFrame([], columns='单图片地址'.split(','))                        
        df.to_excel(writer, sheet_name=page, index=False)
    